package com.ocom.oauth.config;

import com.anji.captcha.service.CaptchaCacheService;
import com.ocom.redis.util.RedisUtil;
import org.springframework.core.annotation.Order;

import javax.annotation.Resource;

/**
 * 对于分布式部署的应用，我们建议应用自己实现CaptchaCacheService，比如用Redis，参考service/spring-boot代码示例。
 * 如果应用是单点的，也没有使用redis，那默认使用内存。
 * 内存缓存只适合单节点部署的应用，否则验证码生产与验证在节点之间信息不同步，导致失败。
 * <p>
 * ☆☆☆ SPI： 在resources目录新建META-INF.services文件夹(两层)，参考当前服务resources。
 *
 * @author Zhuhui
 * @Title: 使用redis缓存
 * @date 2025年7月30日10:13:27
 */
@Order(10)   //要晚于redis的加载后
public class CaptchaCacheServiceRedisImpl implements CaptchaCacheService {


    @Resource
    private RedisUtil redisUtil;

    @Override
    public String type() {
        return "redis";
    }

    private static final String LUA_SCRIPT = "local key = KEYS[1] " +
            "local incrementValue = tonumber(ARGV[1]) " +
            "if redis.call('EXISTS', key) == 1 then " +
            "    return redis.call('INCRBY', key, incrementValue) " +
            "else " +
            "    return incrementValue " +
            "end";


    @Override
    public void set(String key, String value, long expiresInSeconds) {
        if(key.startsWith("AJ.CAPTCHA.REQ.LIMIT-GET") ||  key.startsWith("AJ.CAPTCHA.REQ.LIMIT-CHECK")
                ||  key.startsWith("AJ.CAPTCHA.REQ.LIMIT-FAIL") ){
            redisUtil.set(key, Integer.valueOf(value), expiresInSeconds);
            return;
        }
        redisUtil.set(key, value, expiresInSeconds);
    }

    @Override
    public boolean exists(String key) {
        Object ob = redisUtil.get(key);
        return ob == null ? false : true;
    }

    @Override
    public void delete(String key) {
        redisUtil.del(key);
    }

    @Override
    public String get(String key) {
        Object ob = redisUtil.get(key);
        return ob == null ? null :  ob.toString();
    }

    @Override
    public Long increment(String key, long val) {
        return redisUtil.incr(key, val);
    }

    @Override
    public void setExpire(String key, long l) {
        redisUtil.expire(key, l);
    }
}
